Gated DeltaNet-2: Decoupling Erase and Write in Linear Attention
Gated DeltaNet-2: 线性注意力中擦除与写入的解耦
🧭 线性注意力 · 记忆更新解耦
NVIDIA
将 Delta 规则的记忆更新拆分为通道级擦除门与写入门,突破原有标量门控的耦合限制,显著提升长上下文检索性能。
📌 工作参考价值(4 点)
- 超长用户序列建模:电商中用户行为序列可达数万步,Gated DeltaNet-2 的线性时间复杂度和常量级解码内存可直接用于高效序列编码,其通道级擦除/写入解耦能更精细地更新兴趣记忆,避免旧兴趣过早丢失或新兴趣写入不足。
- 记忆网络的特征交互:可将该方法的核心思想——解耦“遗忘”与“记忆”的强度控制——迁移到特征交互的记忆模块,例如在 Deep Interest Network 中,用独立的通道门控来更新交叉特征的记忆,提升对用户长期偏好的捕获。
- 并行化训练的工程借鉴:论文推导的 chunkwise WY 算法和门控感知反向传播可直接用于大规模推荐模型的序列训练,在不牺牲模型表达能力的前提下大幅提高训练吞吐,适合线上内容快速迭代。
- Agent 工作记忆的维护:在多智能体系统中,该技术可用于维护上下文压缩后的记忆状态,通过解耦擦除和写入,实现对对话历史、环境状态的精准增量编辑,增强 Agent 在长对话或复杂任务中的上下文跟踪能力。